package com.gopay.common.util;


import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.ChannelSftp.LsEntry;
import com.jcraft.jsch.SftpException;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Vector;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * sftp工具类
 * @SftpUtil.java
 * @author dong-wang5@hnari.com
 * @2016年7月15日 上午11:04:24  www.gopay.com.cn Inc.All rights reserved.
 */
public class SftpUtil {

	/** logger */
	private static final Logger LOGGER = LoggerFactory.getLogger(SftpUtil.class);

	/**
	 * 上传文件
	 *
	 * @param directory
	 *            上传的目录，支持多层目录
	 * @param uploadFile
	 *            要上传的文件
	 * @param sftp
	 */
	public static void upload(String directory, String uploadFile, ChannelSftp sftp) {
		String[] split = directory.split("/");
		for (String s : split) {
			try {
				sftp.cd(s);
			} catch (SftpException sException) {
				if(sftp.SSH_FX_NO_SUCH_FILE == sException.id){
					try {
						sftp.mkdir(s);
						sftp.cd(s);
					} catch (SftpException e) {
						LOGGER.error("=====create folder error>"+e.getMessage(),e);
					}
				}
			}
		}
		FileInputStream is = null;
		try {
			LOGGER.info("=====Current path is>" + sftp.pwd());
			File file = new File(uploadFile);
			is = new FileInputStream(file);
			sftp.put(is, file.getName());
			LOGGER.info("=====sftp upload success>>" + file.getName());
		} catch (Exception e) {
			LOGGER.error("=====sftp upload error>"+e.getMessage(),e);
		}finally {
			if(is !=null) {
				try {
					is.close();
				} catch (IOException e) {
					LOGGER.error("FileInputStream close error>>"+e.getMessage(),e);
				}
			}
		}
	}

	/**
	 * 上传文件  add by jianghp 20170710
	 * @param directory
	 * @param uploadFile
	 * @param sftp
	 * @param newName
	 */
	public static void upload(String directory, String uploadFile, ChannelSftp sftp,String newName) {
		String[] split = directory.split("/");
		for (String s : split) {
			try {
				sftp.cd(s);
			} catch (SftpException sException) {
				if(sftp.SSH_FX_NO_SUCH_FILE == sException.id){
					try {
						sftp.mkdir(s);
						sftp.cd(s);
					} catch (SftpException e) {
						LOGGER.error("=====create folder error>"+e.getMessage(),e);
					}
				}
			}
		}
		FileInputStream is = null;
		try {
			LOGGER.info("=====Current path is>" + sftp.pwd()+";NewName="+newName);
			File file = new File(uploadFile);
			is = new FileInputStream(file);
			sftp.put(is, newName);
			LOGGER.info("=====sftp upload success>>" + newName);
		} catch (Exception e) {
			LOGGER.error("=====sftp upload error>"+e.getMessage(),e);
		}finally {
			if(is !=null) {
				try {
					is.close();
				} catch (IOException e) {
					LOGGER.error("FileInputStream close error>>"+e.getMessage(),e);
				}
			}
		}

	}

	/**
	 * 下载文件
	 *
	 * @param directory
	 *            下载目录
	 * @param downloadFile
	 *            下载的文件
	 * @param saveFile
	 *            存在本地的路径
	 * @param sftp
	 */
	@Deprecated
	public static boolean download(String directory, String downloadFile,String saveFile, ChannelSftp sftp) {
		boolean result = false;
		FileOutputStream fileOutputStream = null;
		try {
			sftp.cd(directory);
			File file = new File(saveFile);
			fileOutputStream = new FileOutputStream(file);
			sftp.get(downloadFile, fileOutputStream);
			result = true;
		} catch (Exception e) {
			LOGGER.error("【SFTP连接工具】下载文件失败，文件不存在，路径为：{}/{}" , directory , downloadFile);
		}finally {
			try {
				if (fileOutputStream!=null){
					fileOutputStream.close();
				}
			} catch (IOException e) {
				LOGGER.error("【SFTP连接工具】文件流关闭失败！",e);
			}
		}
		return result;
	}

	/**
	 * sftp下载文件（包含创建连接）
	 * @param hostName 主机名称
	 * @param port 端口号
	 * @param userName 用户名
	 * @param pwd 密码
	 * @param timeout 超时时间
	 * @param directory 下载目录
	 * @param downloadFile 下载的文件
	 * @param saveFolder 存在本地的路径
	 * @return boolean
	 */
	public static boolean download(String hostName, int port, String userName, String pwd, int timeout,
			String directory, String downloadFile,String saveFolder) {
		boolean result = false;
		ChannelSftp chSftp = null;
		FileOutputStream fileOutputStream = null;
		try {
			chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
			chSftp.cd(directory);
			if (!saveFolder.endsWith("/")){
				saveFolder = saveFolder + "/";
			}
			File file = new File(saveFolder + downloadFile);
			fileOutputStream = new FileOutputStream(file);
			chSftp.get(downloadFile, fileOutputStream);
			result = true;
		} catch (Exception e) {
			LOGGER.error("【SFTP连接工具】下载文件失败，文件不存在，文件夹路径：{}，文件名：{}",directory,downloadFile);
		} finally {
			try {
				if (fileOutputStream!=null){
					fileOutputStream.close();
				}
				if (chSftp!=null){
					chSftp.quit();
				}
				SftpChannel.closeChannel();
			} catch (Exception ee) {
				LOGGER.error("【SFTP连接工具】管道关闭失败！" , ee);
			}
		}
		return result;
	}
	
	/**
	 * sftp判断是否存在文件(包含创建连接)
	 * @param hostName 主机名称
	 * @param port	端口号
	 * @param userName	用户名
	 * @param pwd	密码
	 * @param timeout	超时时间
	 * @param directory	下载目录
	 * @param downloadFile 下载文件
	 * @return
	 */
	public static boolean exist(String hostName, int port, String userName, String pwd, int timeout,
			String directory, String downloadFile) {
		boolean result = false;
		ChannelSftp chSftp = null;
		try {
			chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
			Vector<LsEntry> v = chSftp.ls(directory);
			for (LsEntry lsEntry : v) {
				if(lsEntry.getFilename().equals(downloadFile)){
					result = true;
					break;
				}
			}
			LOGGER.error("查询文件: " + downloadFile + "是否存在: " + result);
			
		} catch (Exception e) {
			LOGGER.error("file exist failed！！！！！！", e);
		} finally {
			try {
				chSftp.quit();
				SftpChannel.closeChannel();
			} catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
			}
		}
		return result;
	}
	
	/**
	 * sftp按照名称模糊下载文件并返回文件名（包含创建连接）
	 * @param hostName 主机名称
	 * @param port 端口号
	 * @param userName 用户名
	 * @param pwd 密码
	 * @param timeout 超时时间
	 * @param directory 下载目录
	 * @param downloadFile 下载的文件（XXXX.扩展名，如111.txt）
	 * @param saveFilePath 存在本地的路径
	 */
	public static void downloadLikeName(String hostName, int port, String userName, String pwd, int timeout,
			String directory, String downloadFile,String saveFilePath) {
		ChannelSftp chSftp = null;
		String downloadFileName = null;//待下载的文件全名
		try {
			chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
			Vector<LsEntry> v = chSftp.ls(directory);
			for (LsEntry lsEntry : v) {
				if(lsEntry.getFilename().contains(downloadFile.substring(0,downloadFile.indexOf(".")))){
					downloadFileName = lsEntry.getFilename();
					break;
				}
			}
			if(StringUtils.isEmpty(downloadFileName)){
				return ;
			}
			chSftp.cd(directory);
			File file = new File(saveFilePath + downloadFile);
			chSftp.get(downloadFileName, new FileOutputStream(file));
			
		} catch (Exception e) {
			LOGGER.error("=====sftp download fail:"+e.getMessage());
			e.printStackTrace();
		} finally {
			try {
				chSftp.quit();
				SftpChannel.closeChannel();
			} catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
			}
		}
	}
	
	/**
	 * sftp判断是否存在文件(包含创建连接)(模糊查询文件名)
	 * @param hostName 主机名称
	 * @param port	端口号
	 * @param userName	用户名
	 * @param pwd	密码
	 * @param timeout	超时时间
	 * @param directory	下载目录
	 * @param downloadFile 下载文件名称（XXXX.扩展名，如111.txt）
	 * @return
	 */
	public static boolean existLikeName(String hostName, int port, String userName, String pwd, int timeout,
			String directory, String downloadFile) {
		boolean result = false;
		ChannelSftp chSftp = null;
		try {
			chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
			Vector<LsEntry> v = chSftp.ls(directory);
			for (LsEntry lsEntry : v) {
				if(lsEntry.getFilename().contains(downloadFile.substring(0,downloadFile.indexOf(".")))){
					result = true;
					break;
				}
			}
			LOGGER.error("查询文件: " + downloadFile + "是否存在: " + result);
			
		} catch (Exception e) {
			LOGGER.error("file exist failed！！！！！！", e);
		} finally {
			try {
				chSftp.quit();
				SftpChannel.closeChannel();
			} catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
			}
		}
		return result;
	}
	
	/**
	 * 删除文件
	 *
	 * @param directory
	 *            要删除文件所在目录
	 * @param deleteFile
	 *            要删除的文件
	 * @param sftp
	 */
	public static void delete(String directory, String deleteFile, ChannelSftp sftp) {
		try {
			sftp.cd(directory);
			sftp.rm(deleteFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
     * sftp下载文件（包含创建连接）
     * @param hostName 主机名称
     * @param port 端口号
     * @param userName 用户名
     * @param pwd 密码
     * @param timeout 超时时间
     * @param directory 下载目录
     * @param downloadFile 下载的文件
     * @param saveFile 存在本地的路径
     */
    public static void download2(String hostName, int port, String userName, String pwd, int timeout,
            String directory, String downloadFile,String saveFile) {
        ChannelSftp chSftp = null;
        try {
            chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
            chSftp.cd(directory);
            File file = new File(saveFile);
            chSftp.get(downloadFile, new FileOutputStream(file));

        } catch (Exception e) {
			LOGGER.error("=====sftp download fail:"+e.getMessage());
            e.printStackTrace();
        } finally {
            try {
                chSftp.quit();
                SftpChannel.closeChannel();
            } catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
            }
        }
    }
    
    /** 
     * 读取sftp服务器指定文件
     * @param hostName  主机
     * @param port  端口
     * @param userName 用户名
     * @param pwd  密码
     * @param timeout 超时时间
     * @param directory 目录
     * @param readFile  待读取的文件
     * @param encoding  编码方式
     * @return 
     */
    public static String readFile(String hostName, int port, String userName, String pwd, int timeout,
            String directory,String readFile,String encoding){
        ChannelSftp chSftp = null;
        InputStream ins = null;
        StringBuilder builder = null;
        try {
            chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
            chSftp.cd(directory);
            ins = chSftp.get(readFile);
            BufferedReader reader = new BufferedReader(new InputStreamReader(ins,encoding));
            String line;
            builder = new StringBuilder(150);            
            while ((line = reader.readLine()) != null) {
                builder.append(line);
            }
			LOGGER.error("****************:" + builder.toString());
            reader.close();
            if (ins != null) {
                ins.close();
            }
        } catch (Exception e) {
			LOGGER.error("=====sftp readFile fail:"+e.getMessage());
            e.printStackTrace();
        } finally {
            try {
                chSftp.quit();
                SftpChannel.closeChannel();
            } catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
            }
        }
        return builder.toString();
        
    }

    /**
     * sftp判断是否存在文件(包含创建连接)
     * @param hostName 主机名称
     * @param port  端口号
     * @param userName  用户名
     * @param pwd   密码
     * @param timeout   超时时间 (毫秒)
     * @param directory 下载目录
     * @param downloadFile 下载文件
     * @return
     */
    public static boolean exist2(String hostName, int port, String userName, String pwd, int timeout,
            String directory, String downloadFile) {
        boolean result = false;
        ChannelSftp chSftp = null;
        try {
            chSftp = SftpChannel.getChannel(hostName,port,userName,pwd,timeout);
            Vector<LsEntry> v = chSftp.ls(directory);
            for (LsEntry lsEntry : v) {
                if(lsEntry.getFilename().equals(downloadFile)){
                    result = true;
                    break;
                }
            }
			LOGGER.error("查询文件: " + downloadFile + "是否存在: " + result);

        } catch (Exception e) {
			LOGGER.error("file exist failed！！！！！！", e);
        } finally {
            try {
                chSftp.quit();
                SftpChannel.closeChannel();
            } catch (Exception ee) {
				LOGGER.error("=====sftp close fail:"+ee.getMessage());
            }
        }
        return result;
    }

	/**
	 * 列出目录下的文件
	 *
	 * @param directory
	 *            要列出的目录
	 * @param sftp
	 * @return
	 * @throws SftpException
	 */
	@SuppressWarnings("rawtypes")
	public static Vector listFiles(String directory, ChannelSftp sftp)
			throws SftpException {
		return sftp.ls(directory);
	}

}
